package cn.iocoder.yudao.framework.common.util.monitor;

import lombok.experimental.UtilityClass;
import org.slf4j.MDC;

/**
 * 链路追踪工具类
 * 考虑到每个 starter 都需要用到该工具类，所以放到 common 模块下的 util 包下
 *
 * @author 芋道源码
 */
@UtilityClass
public class TracerUtils {
    private static final String MDC_TRACE_ID = "traceId";

    public static String getTraceId() {
        return MDC.get(MDC_TRACE_ID);
    }

    public static void putTraceId(String traceId) {
        MDC.put(MDC_TRACE_ID, traceId);
    }

    public static void removeTraceId() {
        MDC.remove(MDC_TRACE_ID);
    }

}
